package tp1.datos;

import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;

import tp1.presentacion.modelo.Tren;

public class TrenDAO implements DAO {

	@Override
	public void agregar(Object ob) {
		Tren tren = (Tren)ob;
		Conexion con = new Conexion();
		String sql = "insert into trenes (nom_tren,precio_tren) values(?,?)";
		try {
			PreparedStatement ps = con.getConexion().prepareStatement(sql);
			ps.setString(1, tren.getNombre());
			ps.setFloat(2, tren.getPrecio());
			ps.executeUpdate();
		} catch(SQLException sqlEx) {
			sqlEx.printStackTrace();
		} finally {
			try {
				con.getConexion().close();
			} catch (SQLException se) {
				se.printStackTrace();
			}
		}
	}

	@Override
	public ArrayList<Tren> getTodos() {
		ArrayList<Tren> trenes = new ArrayList<Tren>();
		Conexion con = new Conexion();
		String sql = "select * from trenes;";
		try {
			Statement st = con.getConexion().createStatement();
			ResultSet rs = st.executeQuery(sql);
			while(rs.next()) {
				int cod = rs.getInt("cod_tren");
				String nom = rs.getString("nom_tren");
				float pr = rs.getFloat("precio_tren");
				Tren trAct = new Tren();
				trAct.setCodigo(cod);
				trAct.setNombre(nom);
				trAct.setPrecio(pr);
				trenes.add(trAct);
			}
		} catch(SQLException sqlEx) {
			sqlEx.printStackTrace();
		}
		return trenes;
	}

	@Override
	public Tren getUno(int id) {
		Tren tren = new Tren();
		Conexion con = new Conexion();
		String sql = "select * from trenes where cod_tren = ?;";
		try {
			PreparedStatement pst = con.getConexion().prepareStatement(sql);
			pst.setInt(1, id);
			ResultSet rst = pst.executeQuery();
			if(rst.next()) {
				int cod = rst.getInt("cod_tren");
				String nom = rst.getString("nom_tren");
				float pr = rst.getFloat("precio_tren");
				tren.setCodigo(cod);
				tren.setNombre(nom);
				tren.setPrecio(pr);
			}				
		} catch(SQLException sqlEx) {
			sqlEx.printStackTrace();
		}
		return tren;
	}

}
